package com.tangyuan.ai.entity.dto;

import com.tangyuan.dto.PageDto;
import lombok.Data;

@Data
public class RoleMaskQueryDTO extends PageDto {

    /**
     * 面具名称
     */
    private String mask;
    /**
     * 类型 1.文生文模型 2.文生图描述
     */
    private Integer type;
    /**
     * 分类ID
     */
    private Integer maskCategoryId;

    public Integer getType() {
        if(type == null || type == 0){
            return 1;
        }
        return type;
    }
}
